Assign elements to an array in java